# -*- coding: utf-8 -*-
# !/usr/bin/env python
"""
-------------------------------------------------
   File Name：     speedtest
   Description :   
   Author :       lth
   date：          2022/6/27
-------------------------------------------------
   Change Activity:
                   2022/6/27 9:35: create this script
-------------------------------------------------
"""
__author__ = 'lth'

from metric import Timer
from model import UNet
import torch

if __name__=="__main__":
    model=UNet().cuda()
    dummy_input=torch.randn([1,3,800,800]).cuda()
    metr=Timer()
    model(dummy_input)
    metr.tic()
    for i in range(10):
        model(dummy_input)
    print(metr.toc())